"use client";

import { Portal } from "@ark-ui/react/portal";
import { Select, createListCollection } from "@ark-ui/react/select";
import { ChevronDownIcon } from "lucide-react";
import { useState } from "react";

interface Country {
  label: string;
  value: string;
  flag: string;
}

export default function CountrySelector() {
  const [selectedCountry, setSelectedCountry] = useState<string>("us");

  const countries: Country[] = [
    { label: "United States", value: "us", flag: "🇺🇸" },
    { label: "United Kingdom", value: "uk", flag: "🇬🇧" },
    { label: "Canada", value: "ca", flag: "🇨🇦" },
    { label: "Germany", value: "de", flag: "🇩🇪" },
    { label: "France", value: "fr", flag: "🇫🇷" },
    { label: "Japan", value: "jp", flag: "🇯🇵" },
    { label: "Australia", value: "au", flag: "🇦🇺" },
    { label: "Brazil", value: "br", flag: "🇧🇷" },
  ];

  const collection = createListCollection<Country>({ items: countries });

  const selectedCountryData = countries.find(
    (country) => country.value === selectedCountry
  );

  return (
    <div className="bg-white dark:bg-gray-800 w-full px-4 py-12 rounded-xl flex items-center justify-center">
      <div className="max-w-sm w-full">
        <Select.Root
          collection={collection}
          defaultValue={["us"]}
          onValueChange={(e) => setSelectedCountry(e.value[0] || "us")}
        >
          <Select.Label className="text-sm font-medium text-gray-900 dark:text-gray-100 mb-2">
            Country
          </Select.Label>
          <Select.Control>
            <Select.Trigger className="flex h-10 w-full items-center justify-between rounded-lg border border-gray-300 dark:border-gray-600 bg-white dark:bg-gray-800 px-3 py-2 text-sm text-gray-900 dark:text-gray-100 placeholder-gray-500 dark:placeholder-gray-400 focus:border-gray-900 dark:focus:border-gray-100 focus:outline-none focus:ring-1 focus:ring-gray-900 dark:focus:ring-gray-100">
              <div className="flex items-center gap-2">
                {selectedCountryData && (
                  <span className="text-lg">{selectedCountryData.flag}</span>
                )}
                <Select.ValueText placeholder="Select a country" />
              </div>
              <Select.Indicator>
                <ChevronDownIcon className="h-4 w-4 text-gray-500 dark:text-gray-400" />
              </Select.Indicator>
            </Select.Trigger>
            <Select.ClearTrigger className="text-sm text-gray-500 dark:text-gray-400 hover:text-gray-700 dark:hover:text-gray-300 mt-1">
              Clear
            </Select.ClearTrigger>
          </Select.Control>
          <Portal>
            <Select.Positioner>
              <Select.Content className="z-50 min-w-(--reference-width) rounded-lg border border-gray-200 dark:border-gray-700 bg-white dark:bg-gray-800 shadow-lg max-h-60 overflow-auto">
                <Select.ItemGroup>
                  <Select.ItemGroupLabel className="px-3 py-2 text-xs font-medium text-gray-500 dark:text-gray-400 uppercase tracking-wide">
                    Countries
                  </Select.ItemGroupLabel>
                  {collection.items.map((country) => (
                    <Select.Item
                      key={country.value}
                      item={country}
                      className="relative flex cursor-pointer select-none items-center px-3 py-2 text-sm text-gray-900 dark:text-gray-100 data-highlighted:bg-gray-100 dark:data-highlighted:bg-gray-700 data-[state=checked]:bg-gray-50 dark:data-[state=checked]:bg-gray-700"
                    >
                      <span className="mr-3 text-lg">{country.flag}</span>
                      <Select.ItemText>{country.label}</Select.ItemText>
                      <Select.ItemIndicator className="absolute right-3 text-blue-600 dark:text-blue-400">
                        ✓
                      </Select.ItemIndicator>
                    </Select.Item>
                  ))}
                </Select.ItemGroup>
              </Select.Content>
            </Select.Positioner>
          </Portal>
          <Select.HiddenSelect />
        </Select.Root>
      </div>
    </div>
  );
}
